####################################################################################
###
###   Immigration, Public Housing and Support for the French National Front
###   Gloria Gennaro
###
###   Paper Figure A2
###
####################################################################################


library(dplyr)
library(ggplot2)
library(miceadds)
library(AER)
library(rlist)


################################################################################
# Set Up
################################################################################

# Load Working Directories
source(paste0(wd_main, '/2_code/00_working_directories.R'))

# Load data and clean
source(paste0(wd_code, '/01_data_load_clean.R'))


################################################################################
# Variable definition
################################################################################

df_did$imm_dec= cut(df_did$imm_share_99,
                          breaks=c(quantile(df_did$imm_share_99, probs = seq(0, 1, by = 0.2), na.rm=T)),
                          labels=c("1","2","3", '4', '5'), include.lowest=T)


################################################################################
# Plot Function
################################################################################

makeplot = function(model_output, hete_var){
  groups = c(paste0('Low ', hete_var), paste0('Low-Medium ', hete_var), paste0('Medium ', hete_var), paste0('Medium-High ', hete_var), paste0('High ', hete_var))
  container = lapply(model_output, function(x) rbind(cbind(x[2:6,], period = c(-4, -3, -2, 1, 3)), c(0,0,NA,NA,-1), c(NA,NA,NA,NA,0),  c(NA,NA,NA,NA,2)))
  container = list.rbind(container) %>% as.data.frame()
  container = cbind(container, group=rep(groups, each=8))
  
  colnames(container) = c('beta', 'se', 'tval', 'pval', 'period', 'group')
  container$group.f = factor(container$group, levels = groups)
  ggplot(data=container, aes(y=beta, x=factor(period), colour=group.f, shape = group.f))+
    geom_point(position = position_dodge(width = 0.5), size=3) + 
    geom_errorbar(aes(ymin=beta-1.96*se, ymax=beta+1.96*se), linewidth=1, width=.15, position = position_dodge(width = 0.5)) +
    geom_hline(yintercept=0, colour='red')+
    xlab("Period relative to policy implementation") + 
    ylab("NF Vote Share (percentage points)") +
    theme_light()+
    theme(legend.position="bottom", legend.title = element_blank(), text = element_text(size=18))
}
  

################################################################################
# Estimation
################################################################################

modelli = list()
for (n in c(1,2,3,4,5)){
  
  print(paste0('Immigation quintile ', n, ' -------------- '))
  df1 = df_did[which(!is.na(df_did$fn) & df_did$imm_dec==n),]
  
  temp = lm(data=df1, fn ~  m4 + m3 + m2 + p1 + p2 + p3 + running + 
              factor(CODGEO) + factor(policy_period) )
  temp = coeftest(temp, vcov = vcovCL, cluster = ~df1$CODGEO)
  
  print(temp[1:10,])
  modelli[[n]] = temp
}


################################################################################
# Plot (Figure 2)
################################################################################

makeplot(modelli, 'Immigration')
ggsave(paste0(wd_res, '/figures/figA2.pdf'), height=7, width=10)



